package com.birdboot.http;

import javax.activation.MimetypesFileTypeMap;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * 响应对象
 * 该类的每一个实列用于表示HTTP协议规定的服务端给客户端发送的响应内容
 * 状态行，响应头，响应正文
 */
public class HttpServletResponse {
    // v10改动：将MimetypesFileTypeMap定义为静态的，全局唯一，重复利用
    private static MimetypesFileTypeMap mftm = new MimetypesFileTypeMap();
    private Socket socket;
    // 状态行相关信息
    private int statusCode = 200;
    private String statusReason = "OK";
    // 响应头相关信息
    // v9新增内容：添加一个Map类型的属性用于保存所有有待发送的响应头
    private Map<String,String > headers = new HashMap<>();
    // 响应正文相关信息
    private File contentFile;// 正文对应文件

    public HttpServletResponse(Socket socket){
        this.socket = socket;
    }
    // 新增响应方法，内容就是原ClientHandler里第三步发送响应的代码
    public void response() throws IOException {
        // 将页面包含在响应中发送给浏览器
        /**
         * HTTP/1.1 404 NotFound(CRLF)
         * Content-Type: text/html(CRLF)
         * Content-Length: 2546(CRLF)(CRLF)
         * 1011101010101010101......
         */
        // 1发送状态行
        sendStatusLine();
        // 2发送响应头
        sendHeaders();
        // 3发送响应正文
        sendContent();
    }
    // 1发送状态行
    private void sendStatusLine() throws IOException {

        println("HTTP/1.1"+" "+statusCode+" "+statusReason);
    }
    // 2发送响应头
    private void sendHeaders() throws IOException {
        // 通过遍历Headers的Map，将所有待发送的响应头发送给浏览器
        Set<Map.Entry<String,String>> entrySet = headers.entrySet();
        for (Map.Entry<String ,String> e : entrySet) {
            String key = e.getKey();
            String value = e.getValue();
            println(key+": "+value);
        }
//        println("Content-Type: text/html");
//        println("Content-Length: "+contentFile.length());
        // 单独发送回车加换行表达响应头发完了
        println("");
    }
    // 3发送响应正文
    private void sendContent() throws IOException {

        OutputStream out = socket.getOutputStream();
        FileInputStream fis = new FileInputStream(contentFile);
        byte[] data = new byte[1024*10];
        int len;
        while ((len=fis.read(data))!=-1){
            out.write(data,0,len);
        }
    }

    private void println(String line) throws IOException {
        OutputStream out = socket.getOutputStream();
        out.write(line.getBytes(StandardCharsets.ISO_8859_1));
        out.write(13);
        out.write(10);
    }
    public void addHeader(String name,String value){
        headers.put(name,value);
    }

    public int getStatusCode() {
        return statusCode;
    }

    public void setStatusCode(int statusCode) {
        this.statusCode = statusCode;
    }

    public String getStatusReason() {
        return statusReason;
    }

    public void setStatusReason(String statusReason) {
        this.statusReason = statusReason;
    }

    public File getContentFile() {
        return contentFile;
    }

    public void setContentFile(File contentFile) {
        this.contentFile = contentFile;
        //V10改动，将添加响应头Content-Type和Content-Length操作移动到设置正文方法中，应为只要有正文，就要设置文件类型和文件长度

        String contentType = mftm.getContentType(contentFile);
        addHeader("Content-Type",contentType);
        addHeader("Content-Length",contentFile.length()+"");
    }
}